#include "problems/1.h"
#include "gtest/gtest.h"

TEST(TestProblemSolutions, TestProblem1)
{
    problem_1::Solution s;
    vector<int> nums{2, 7, 11, 15};
    ASSERT_EQ(s.twoSum(nums, 9), vector<int>({0, 1}));
    nums = {3, 2, 4};
    ASSERT_EQ(s.twoSum(nums, 6), vector<int>({1, 2}));
}